import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
from agent import WindowAgent
import time

print("=== 光标输入示例 ===")

def main():
    agent = WindowAgent()
    
    if agent.start_browser("chrome"):
        try:
            # 打开百度
            print("打开百度...")
            agent.browser_navigate("https://www.baidu.com")
            time.sleep(3)
            
            # 点击搜索框位置（根据屏幕调整坐标）
            print("点击搜索框...")
            # if agent.browser_click_position(960, 400):  # 屏幕中央偏上
            #     print("点击完成")
            time.sleep(2)
            
            # 直接在光标位置输入
            print("在光标位置输入...")
            if agent.browser_type_at_cursor("Python自动化"):
                print("输入完成")
            time.sleep(2)
            
            # 按回车搜索
            print("按回车搜索...")
            agent.send_keys('enter')
            time.sleep(3)
            
            # 截屏
            if agent.browser_screenshot("cursor_search_result.png"):
                print("截屏成功")
            
            # 等待查看结果
            print("\n等待 5 秒查看结果...")
            time.sleep(5)
                
        except Exception as e:
            print(f"错误: {e}")
            import traceback
            traceback.print_exc()
        finally:
            print("\n准备关闭浏览器...")
            time.sleep(3)
            agent.close_browser()
            print("浏览器已关闭")

if __name__ == "__main__":
    main()
